Tensorflow Tutorial 15 - Customizing Model.fit